python conditional for loop